package com.zixi.ai.common.enums;

import lombok.Getter;

/**
 * AI 模型常量类
 * @author wfg
 */
@Getter
public enum AiModelEnum {
    /**
     * 智谱文生图服务
     */
    ZHI_PU_MODEL_COG_VIEW3_PLUS("cogview-3-plus", "适用于图像生成任务"),
    LLAMA_3_2_7B_8K("llama3.2", "llama3.2"),
    QWEN_2_0_5B("qwen2.5:1.5b", "qwen2.5:1.5b");
    private final String code;
    private final String desc;

    AiModelEnum(String code, String desc) {
        this.code = code;
        this.desc = desc;
    }

    @Override
    public String toString() {
        return "AiModelEnum{" +
                "code='" + code + '\'' +
                ", desc='" + desc + '\'' +
                '}';
    }
}
